import torch
import torch.nn as nn
from torchvision import datasets, transforms
from torchvision import models

# model = datasets.MNIST('./data', download=True)
# model = models.resnet18(pretrained=True)
# model1 = models.resnet34(pretrained=True)
# model2 = models.resnet101(pretrained=True)
# model3 = models.resnet152(pretrained=True)
#
# model4 = models.vgg11(pretrained=True)
# model41 = models.vgg13(pretrained=True)
# model42 = models.vgg16(pretrained=True)
# model43 = models.vgg19(pretrained=True)
# model4134 = models.vgg11_bn(pretrained=True)
# model411 = models.vgg13_bn(pretrained=True)
# model421 = models.vgg16_bn(pretrained=True)
# model431 = models.vgg19_bn(pretrained=True)